'''
    作者：刘妍
    版本：4.0
    日期：22/05/2020
    1.0功能：计算空气质量指数AQI
    2.0新增功能：读取已经获取的JSON数据文件，并将AQI前5的数据输出到文件
    3.0新增功能：读取已经获取的JSON数据文件，并将其转换为CSV文件
    4.0新增功能：自动识别文件类型（JSON文件还是CSV文件）
'''

import json
import csv
import os

# 自定义函数，处理JSON文件
def process_json_file(filepath):
    with open(filepath, mode='r', encoding='utf-8') as f:
        city_list = json.load(f)
    print(city_list)


# 自定义函数，处理CSV文件
def process_csv_file(filepath):
    with open(filepath, mode='r', encoding='utf-8') as f:
        reader = csv.reader(f)
        for row in reader:
            print(','.join(row))


def main():
    filepath = input('请输入文件名称：')
    # 使用os模块中的函数对文件名进行分割
    file_name, file_ext = os.path.splitext(filepath)

    if file_ext == '.json':
        process_json_file(filepath)
    elif file_ext == '.csv':
        process_csv_file(filepath)
    else:
        print('不支持的文件格式！')


if __name__ == '__main__':
    main()